package net.sf.rose.file.excel.poi.model;

import java.util.ArrayList;
import java.util.List;

/**
 * @author fengjian
 * @since 2016/05/08
 */
public class BookModel {

    /** 工作薄名称 */
    private String name = null;

    /** 表数据模型列表 */
    private List<SheetModel> sheetModels = new ArrayList<>();

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public List<SheetModel> getSheetModels() {
        return sheetModels;
    }

    public SheetModel getSheetModel(int index) {
        return sheetModels.get(index);
    }

    public SheetModel getSheetModel(String sheetName) {
        for (SheetModel sheet : sheetModels) {
            if (sheet.getName().equals(sheetName)) {
                return sheet;
            }
        }
        return null;
    }

    public void setSheetModels(List<SheetModel> sheetModels) {
        this.sheetModels = sheetModels;
    }

    public void addSheetModel(SheetModel sheetModel) {
        this.sheetModels.add(sheetModel);
    }
}
